# 应用聚合函数
import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randn(5, 4), index=pd.date_range('12/14/2020', periods=5), columns=['A', 'B', 'C', 'D'])
print(df)
# 窗口大小为3，min_periods 最小观测值为1
r = df.rolling(window=3, min_periods=1)
print(r)
print(r.aggregate(np.sum))  # 使用 aggregate()聚合操作
print(r['A'].aggregate(np.sum))  # 对 A 列聚合
print(r['A', 'B'].aggregate(np.sum))  # 对 A/B 两列聚合
print(r['A', 'B'].aggregate([np.sum, np.mean]))  # 对 A/B 两列聚合
print(r.aggregate({'A': np.sum, 'B': np.mean}))  # 对不同列应用不同函数
